<?php

namespace app\admin\controller;

use app\admin\model\InvestmentCategoryModel;
use app\api\service\cms\AttachmentService;
use app\api\service\cms\CacheService;
use app\api\service\cms\CutWordIndex;
use app\common\controller\Backend;
use app\common\model\Attachment;
use PhpOffice\PhpSpreadsheet\Spreadsheet;
use PhpOffice\PhpSpreadsheet\Writer\Xlsx;
use PhpOffice\PhpSpreadsheet\Helper\Html as HtmlHelper;
use think\Db;
use think\db\exception\DataNotFoundException;
use think\db\exception\ModelNotFoundException;
use think\exception\DbException;
use think\exception\PDOException;
use think\exception\ValidateException;

/**
 *
 *
 * @icon fa fa-circle-o
 */
class Investment extends Backend
{

    /**
     * Investment模型对象
     * @var \app\admin\model\Investment
     */
    protected $model = null;

    public function _initialize()
    {
        parent::_initialize();
        $this->model = new \app\admin\model\Investment;
        $categoryList = InvestmentCategoryModel::field('id,platename,type')->select();//分类数据
        $this->view->assign("categoryList", $categoryList);

    }

    /**
     * 默认生成的控制器所继承的父类中有index/add/edit/del/multi五个基础方法、destroy/restore/recyclebin三个回收站方法
     * 因此在当前控制器中可不用编写增删改查的代码,除非需要自己控制这部分逻辑
     * 需要将application/admin/library/traits/Backend.php中对应的方法复制到当前控制器,然后进行修改
     */


    /**
     * 查看
     */
    public function index()
    {
        //当前是否为关联查询
        $this->relationSearch = true;
        //设置过滤方法
        $this->request->filter(['strip_tags', 'trim']);
        if ($this->request->isAjax()) {
            //如果发送的来源是Selectpage，则转发到Selectpage
            if ($this->request->request('keyField')) {
                return $this->selectpage();
            }
            list($where, $sort, $order, $offset, $limit) = $this->buildparams();

            $list = $this->model
                ->with(['admin'])
                ->where($where)
                ->order($sort, $order)
                ->paginate($limit);
            foreach ($list as &$v){
                //去除html标签,截取字符串
                $v['macro_content'] = mb_substr(strip_tags($v['macro_content']), 0, 20);
                if(mb_strlen($v['macro_content']) >= 20) {
                    $v['macro_content'] .= '...';
                }
            }
            $result = array("total" => $list->total(), "rows" => $list->items());


            return json($result);
        }
        return $this->view->fetch();
    }

    public function import()
    {
        return parent::import();
    }

    public function export()
    {
        ob_clean();
        $data = $this->model->whereIn('id', json_decode($this->request->param('data'), true))->select();
        $spreadsheet = new Spreadsheet();
        $defaultStyle = $spreadsheet->getDefaultStyle();
        $defaultStyle->getAlignment()->setHorizontal(\PhpOffice\PhpSpreadsheet\Style\Alignment::HORIZONTAL_CENTER)->setVertical(\PhpOffice\PhpSpreadsheet\Style\Alignment::VERTICAL_CENTER);
        $spreadsheet->createSheet();//创建sheet
        $sheet = $spreadsheet->setActiveSheetIndex(0);//设置当前的活动sheet
        $sheet->setTitle('投研日志');  //设置当前sheet的标题
        $mapping = [
            '时间' => 'content_time',
            '宏观面' => 'macro_content',
            '政策面' => 'policy_content',
            '资金面及预期' => 'fund_content',
            '情绪面' => 'emotion_content',
            '市场预期/关注' => 'expect_content',
            '市场新闻' => 'news_content',
            '创建时间' => 'createtime',
            '更新时间' => 'updatetime',
        ];

        $result = [];

        foreach ($mapping as $index => $value) {
            $result[] = [
                'key' => $index,
                'value' => $value
            ];
        }
        //excel表头
        foreach ($result as $key => $val) {
            $spreadsheet->getActiveSheet()->getColumnDimension(getTag($key + 1))->setWidth(20);
            $spreadsheet->getActiveSheet()->getStyle(getTag($key + 1) . '1')->getFont()->setBold(true);
            $spreadsheet->getActiveSheet()->getStyle(getTag($key + 1) . '1')->getFill()->getEndColor()->setRGB('FFC7CE');
            $sheet->getRowDimension(1)->setRowHeight(30);
            $sheet->setCellValue(getTag($key + 1) . '1', $val['key']);
        }

        //excel表主体内容
        foreach ($data as $k => $v) {
            foreach ($result as $m => $n) {
                $cell = $v[$n['value']] ?? '';
                if (in_array($n['value'], ['content_time', 'createtime', 'updatetime'])) {
                    $cell = date('Y-m-d H:i:s', $cell);
                } else {
                    $wizard = new HtmlHelper();
                    $cell = $wizard->toRichTextObject($cell);
                }
                $sheet->setCellValueExplicit(getTag($m + 1) . ($k + 2), $cell, \PhpOffice\PhpSpreadsheet\Cell\DataType::TYPE_STRING);
            }
        }
        header('Content-Type: application/vnd.ms-excel');
        header('Content-Disposition: attachment;filename=投研日志' . date('Y-m-d H:i:s') . '.xlsx');
        header('Cache-Control: max-age=0');
        $writer = new Xlsx($spreadsheet);
        $writer->save('php://output');
    }

    /**
     * 添加
     *
     * @return string
     * @throws \think\Exception
     */
    public function add()
    {
        if (false === $this->request->isPost()) {
            return $this->view->fetch();
        }
        $params = $this->request->post('row/a');
        if (empty($params)) {
            $this->error(__('Parameter %s can not be empty', ''));
        }
        $params = $this->preExcludeFields($params);

        if ($this->dataLimit && $this->dataLimitFieldAutoFill) {
            $params[$this->dataLimitField] = $this->auth->id;
        }
        $result = false;

        //组装插入文件列表的数据
        if ($params['file_list']){
            $file_list = explode(',',$params['file_list']);
            $file_arr = [];
            foreach($file_list as $v){
                $filename = Attachment::where('url',$v)->value('filename');
                $file_arr[] = [
                    'name'=>$filename,
                    'url'=>$v,
                ];
            }
            if ($file_arr){
                $params['file_list'] = json_encode($file_arr, JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT);
            }
        }

        Db::startTrans();
        try {
            //是否采用模型验证
            if ($this->modelValidate) {
                $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.add' : $name) : $this->modelValidate;
                $this->model->validateFailException()->validate($validate);
            }
            $result = $this->model->allowField(true)->save($params);

            //加入缓存
            $cache_service = new CacheService();
            $cache_service->addHashData('investment',$this->model->id);

            //操作分词索引表
            $cut_word_index = new CutWordIndex();
            $cut_word_index->addCutWordIndex($this->model->id,'investment');

            Db::commit();
        } catch (ValidateException|PDOException|Exception $e) {
            Db::rollback();
            $this->error($e->getMessage());
        }
        if ($result === false) {
            $this->error(__('No rows were inserted'));
        }
        $this->success();
    }

    /**
     * 编辑
     *
     * @param $ids
     * @return string
     * @throws DbException
     * @throws \think\Exception
     */
    public function edit($ids = null)
    {
        $row = $this->model->get($ids);
        if (!$row) {
            $this->error(__('No Results were found'));
        }
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds) && !in_array($row[$this->dataLimitField], $adminIds)) {
            $this->error(__('You have no permission'));
        }
        if (false === $this->request->isPost()) {
            rmdirs(TEMP_PATH, false);//清除模板缓存，不然include没办法显示最新的信息
            $row = $row->toArray();

            $row['file_video_url'] = [];
            if (!empty($row['file_video'])) {
                //查询附件表
                $file_video_url = (new AttachmentService())->getList(['url'=>$row['file_video']]);
                foreach ($file_video_url as $k => $v) {
                    $url = obsurl($v['url']);
                    $row['file_video_url'][] = [
                        'url' => $url,
                        'name' => $v['filename'],
                        'is_image' => isImage($url),
                    ];
                }
            }
            $row['file_video_url'] = json_encode($row['file_video_url'], JSON_UNESCAPED_UNICODE);

            $row['file_audio_url'] = [];
            if (!empty($row['file_audio'])) {
                //查询附件表
                $file_audio_url = (new AttachmentService())->getList(['url'=>$row['file_audio']]);
                foreach ($file_audio_url as $k => $v) {
                    $url = obsurl($v['url']);
                    $row['file_audio_url'][] = [
                        'url' => $url,
                        'name' => $v['filename'],
                        'is_image' => isImage($url),
                    ];
                }
            }
            $row['file_audio_url'] = json_encode($row['file_audio_url'], JSON_UNESCAPED_UNICODE);

            $row['file_data_url'] = [];
            if (!empty($row['file_data'])) {
                //查询附件表
                $file_data_url = (new AttachmentService())->getList(['url'=>$row['file_data']]);
                foreach ($file_data_url as $k => $v) {
                    $url = obsurl($v['url']);
                    $row['file_data_url'][] = [
                        'url' => $url,
                        'name' => $v['filename'],
                        'is_image' => isImage($url),
                    ];
                }
            }
            $row['file_data_url'] = json_encode($row['file_data_url'], JSON_UNESCAPED_UNICODE);

            //将json中的url转成字符串
            $row['file_list_url'] = [];
            if (!empty($row['file_list'])){
                $file_list_url = json_decode($row['file_list'],true);
                foreach($file_list_url as $v){
                    $url = obsurl($v['url']);
                    $row['file_list_url'][] = [
                        'url' => $url,
                        'name' => $v['name'],
                        'is_image' => isImage($url),
                    ];
                }
            }
            $row['file_list_url'] = json_encode($row['file_list_url'], JSON_UNESCAPED_UNICODE);

            $this->view->assign('row', $row);
            return $this->view->fetch();
        }
        $params = $this->request->post('row/a');
        if (empty($params)) {
            $this->error(__('Parameter %s can not be empty', ''));
        }
        $params = $this->preExcludeFields($params);
        $result = false;

        //组装插入文件列表的数据
        if ($params['file_list']){
            $file_list = explode(',',$params['file_list']);
            $file_arr = [];
            foreach($file_list as $v){
                $filename = Attachment::where('url',$v)->value('filename');
                $file_arr[] = [
                    'name'=>$filename,
                    'url'=>$v,
                ];
            }
            if ($file_arr){
                $params['file_list'] = json_encode($file_arr, JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT);
            }
        }

        Db::startTrans();
        try {
            //是否采用模型验证
            if ($this->modelValidate) {
                $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.edit' : $name) : $this->modelValidate;
                $row->validateFailException()->validate($validate);
            }
            $result = $row->allowField(true)->save($params);

            //加入缓存
            $cache_service = new CacheService();
            $cache_service->addHashData('investment',$row['id']);

            //操作分词索引表
            $cut_word_index = new CutWordIndex();
            $cut_word_index->updateCutWordIndex($row['id'],'investment');

            Db::commit();
        } catch (ValidateException|PDOException|Exception $e) {
            Db::rollback();
            $this->error($e->getMessage());
        }
        if (false === $result) {
            $this->error(__('No rows were updated'));
        }
        $this->success();
    }

    /**
     * 删除
     *
     * @param $ids
     * @return void
     * @throws DbException
     * @throws DataNotFoundException
     * @throws ModelNotFoundException
     */
    public function del($ids = null)
    {
        if (false === $this->request->isPost()) {
            $this->error(__("Invalid parameters"));
        }
        $ids = $ids ?: $this->request->post("ids");
        if (empty($ids)) {
            $this->error(__('Parameter %s can not be empty', 'ids'));
        }
        $pk = $this->model->getPk();
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds)) {
            $this->model->where($this->dataLimitField, 'in', $adminIds);
        }
        $list = $this->model->where($pk, 'in', $ids)->select();

        $count = 0;
        Db::startTrans();
        try {
            //加入缓存
            $cache_service = new CacheService();
            $cut_word_index = new \app\api\service\cms\CutWordIndex();

            foreach ($list as $item) {
                $cache_service->delHashData('investment', $item->id);//删除缓存

                //操作分词索引表
                $cut_word_index->delCutWordIndex($item->id,'investment');

                $count += $item->delete();
            }
            Db::commit();
        } catch (PDOException|Exception $e) {
            Db::rollback();
            $this->error($e->getMessage());
        }
        if ($count) {
            $this->success();
        }
        $this->error(__('No rows were deleted'));
    }
}
